package com.huyun.web.concurrency;

import lombok.extern.slf4j.Slf4j;
import net.jcip.annotations.NotThreadSafe;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Semaphore;

/**
 * Semaphore翻译成字面意思为 信号量，Semaphore可以阻塞进程并且控制同时访问的线程个数，
 * 通过 acquire() 获取一个许可，如果没有就等待，而 release() 释放一个许可。Semaphore其实和锁有点类似，
 * 它一般用于控制对某组资源的访问权限。
 * CountDownLatch与Semaphore在使用时，通常会与线程池配合使用
 * Semaphore适合控制并发数，CountDownLatch比较适合保证线程执行完后再执行其他处理，因此模拟并发时，
 * 使用两者结合起来是最好的。
 * 。
 */
@Slf4j
@NotThreadSafe //线程不安全的
public class ConcurrencyTest {
    private final Logger logger = LoggerFactory.getLogger(ConcurrencyTest.class);
    // 请求总数
    public static int clientTotal = 5000;

    // 同时并发执行的线程数
    public static int threadTotal = 200;

    public static int count = 0;

    public static void main(String[] args) throws Exception {
        ExecutorService executorService = Executors.newCachedThreadPool();
        //定义信号量
        final Semaphore semaphore = new Semaphore(threadTotal);
        //定义计数器闭锁
        final CountDownLatch countDownLatch = new CountDownLatch(clientTotal);
        Long startTime = System.currentTimeMillis();
        for (int i = 0; i < clientTotal; i++) {
            executorService.execute(() -> {
                try {
                    //获取信号量，否则会阻塞
                    semaphore.acquire();
                    add();
                    semaphore.release();
                } catch (Exception e) {
                    log.error("exception", e);
                }
                //每执行一次则减1
                countDownLatch.countDown();
            });
        }
        Long endTime = System.currentTimeMillis();
        countDownLatch.await();
        //关闭线程池
        executorService.shutdown();

        log.info("count:{}", count);
        log.info("所用时间", (endTime - startTime));
        System.out.println("所用时间" + (endTime - startTime));
    }


    private static void add() {
        count++;
    }
}
